package hc.util

import java.security.*
import javax.crypto.*
import javax.crypto.spec.SecretKeySpec
//https://blog.csdn.net/qq_23888451/article/details/84658360
@groovy.transform.CompileStatic
class AESUtil {
	public static byte[] encrypt(String content, String password) {
		try {
			KeyGenerator kgen = KeyGenerator.getInstance("AES");// 创建AES的Key生产者

//			kgen.init(128, new SecureRandom(password.getBytes()));// 利用用户密码作为随机数初始化出
			SecureRandom random = SecureRandom.getInstance("SHA1PRNG");
			random.setSeed(password.getBytes("utf-8"));
			kgen.init(128, random);
			
			// 128位的key生产者
			//加密没关系，SecureRandom是生成安全随机数序列，password.getBytes()是种子，只要种子相同，序列就一样，所以解密只要有password就行

			SecretKey secretKey = kgen.generateKey();// 根据用户密码，生成一个密钥

			byte[] enCodeFormat = secretKey.getEncoded();// 返回基本编码格式的密钥，如果此密钥不支持编码，则返回
			// null。

			SecretKeySpec key = new SecretKeySpec(enCodeFormat, "AES");// 转换为AES专用密钥

			Cipher cipher = Cipher.getInstance("AES");// 创建密码器

			byte[] byteContent = content.getBytes("utf-8");

			cipher.init(Cipher.ENCRYPT_MODE, key);// 初始化为加密模式的密码器

			byte[] result = cipher.doFinal(byteContent);// 加密

			return result;

		} catch (NoSuchPaddingException e) {
			e.printStackTrace();
		} catch (NoSuchAlgorithmException e) {
			e.printStackTrace();
		} catch (UnsupportedEncodingException e) {
			e.printStackTrace();
		} catch (InvalidKeyException e) {
			e.printStackTrace();
		} catch (IllegalBlockSizeException e) {
			e.printStackTrace();
		} catch (BadPaddingException e) {
			e.printStackTrace();
		}
		return null;
	}
	
	public static byte[] decrypt(byte[] content, String password) {
		try {
			KeyGenerator kgen = KeyGenerator.getInstance("AES");// 创建AES的Key生产者
			
//			kgen.init(128, new SecureRandom(password.getBytes()));
			
			SecureRandom random = SecureRandom.getInstance("SHA1PRNG");
			random.setSeed(password.getBytes("utf-8"));
			kgen.init(128, random);
			
			SecretKey secretKey = kgen.generateKey();// 根据用户密码，生成一个密钥
			byte[] enCodeFormat = secretKey.getEncoded();// 返回基本编码格式的密钥
			SecretKeySpec key = new SecretKeySpec(enCodeFormat, "AES");// 转换为AES专用密钥
			Cipher cipher = Cipher.getInstance("AES");// 创建密码器
			cipher.init(Cipher.DECRYPT_MODE, key);// 初始化为解密模式的密码器
			byte[] result = cipher.doFinal(content);
			return result; // 明文

		} catch (NoSuchAlgorithmException e) {
			e.printStackTrace();
		} catch (NoSuchPaddingException e) {
			e.printStackTrace();
		} catch (InvalidKeyException e) {
			e.printStackTrace();
		} catch (IllegalBlockSizeException e) {
			e.printStackTrace();
		} catch (BadPaddingException e) {
			e.printStackTrace();
		}
		return null;
	}
	
	
	static String key ="aA881bLwMhst6AmfBEkcJYu67VCrkbn1eIz3YyVZKdQ="
	
	public static String encrypt(String content)
	{
		byte[] en = encrypt(content, key);
		return Base64.getEncoder().encodeToString(en);
	}
	
	public static String decrypt(String content)
	{
		byte[] de = Base64.getDecoder().decode(content);
		return new String(decrypt(de, key),"utf-8");
	}
	
	public static void main(String[] args) {
		String content = "美女，约吗？";
//		String password = "123";
//		System.out.println("加密之前：" + content);
//		// 加密
//		byte[] encrypt = AESUtil.encrypt(content, password);
//		System.out.println("加密后的内容：" + new String(encrypt));
//		// 解密
//		byte[] decrypt = AESUtil.decrypt(encrypt, password);
//		System.out.println("解密后的内容：" + new String(decrypt));
		
		println "encrypted:"+encrypt(content);
		println "decrypted:"+decrypt(encrypt(content));
	}
}
